package com.insight.modules.message.util;

import com.alibaba.fastjson.JSONObject;
import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;
import com.insight.modules.message.entity.SysMessage;
import com.insight.modules.message.entity.SysMessageTemplate;
import com.insight.modules.message.handle.enums.SendMsgStatusEnum;
import com.insight.modules.message.service.ISysMessageService;
import com.insight.modules.message.service.ISysMessageTemplateService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.io.StringWriter;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * 消息生成工具
 */

@Component
public class PushMsgUtil {

    @Autowired
    private ISysMessageService sysMessageService;

    @Autowired
    private ISysMessageTemplateService sysMessageTemplateService;

    @Autowired
    private Configuration freemarkerConfig;

    /**
     * 发送消息工具方法
     *
     * @param msgType 消息类型（1短信，2邮件，3微信）
     * @param templateCode 消息模板编码
     * @param map 模板参数，用于填充模板内容
     * @param sentTo 接收消息的目标用户或地址
     * @return true 表示消息保存成功，false 表示失败
     */
    public boolean sendMessage(String msgType, String templateCode, Map<String, String> map, String sentTo) {
        // 根据模板编码查询对应的消息模板
        List<SysMessageTemplate> sysSmsTemplates = sysMessageTemplateService.selectByCode(templateCode);
        SysMessage sysMessage = new SysMessage();
        if (sysSmsTemplates.size() > 0) {
            SysMessageTemplate sysSmsTemplate = sysSmsTemplates.get(0);
            sysMessage.setEsType(msgType);
            sysMessage.setEsReceiver(sentTo);
            // 模板标题
            String title = sysSmsTemplate.getTemplateName();
            // 模板内容
            String content = sysSmsTemplate.getTemplateContent();
            StringWriter stringWriter = new StringWriter();
            try {
                // 使用FreeMarker模板引擎处理模板内容，填充参数
                Template template = new Template("SysMessageTemplate", content, freemarkerConfig);
                template.process(map, stringWriter);
            } catch (IOException | TemplateException e) {
                e.printStackTrace();
                return false;
            }
            content = stringWriter.toString();
            // 设置消息标题、内容、参数等属性
            sysMessage.setEsTitle(title);
            sysMessage.setEsContent(content);
            sysMessage.setEsParam(JSONObject.toJSONString(map));
            sysMessage.setEsSendTime(new Date());
            sysMessage.setEsSendStatus(SendMsgStatusEnum.WAIT.getCode()); // 状态：等待发送
            sysMessage.setEsSendNum(0);
            // 保存消息实体，成功返回true
            if(sysMessageService.save(sysMessage)) {
                return true;
            }
        }
        return false;
    }
}

